package thematic_section.stock_earnings;

/**
 * @ClassName _714_MaxProfit_with_fee_trade
 * @Description: 给定一个整数数组 prices，其中 prices[i]表示第 i 天的股票价格 ；整数 fee 代表了交易股票的手续费用。
 *
 * 你可以无限次地完成交易，但是你每笔交易都需要付手续费。如果你已经购买了一个股票，在卖出它之前你就不能再继续购买股票了。
 *
 * 返回获得利润的最大值。
 *
 * 注意：这里的一笔交易指买入持有并卖出股票的整个过程，每笔交易你只需要为支付一次手续费。
 *
 *  
 *
 * 示例 1：
 *
 * 输入：prices = [1, 3, 2, 8, 4, 9], fee = 2
 * 输出：8
 * 解释：能够达到的最大利润:
 * 在此处买入 prices[0] = 1
 * 在此处卖出 prices[3] = 8
 * 在此处买入 prices[4] = 4
 * 在此处卖出 prices[5] = 9
 * 总利润: ((8 - 1) - 2) + ((9 - 4) - 2) = 8
 * 示例 2：
 *
 * 输入：prices = [1,3,7,5,10,3], fee = 3
 * 输出：6
 *  
 *
 * 提示：
 *
 * 1 <= prices.length <= 5 * 104
 * 1 <= prices[i] < 5 * 104
 * 0 <= fee < 5 * 104
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/best-time-to-buy-and-sell-stock-with-transaction-fee
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * @Author: yongliang.ma
 * @Create_time:2022/10/11 12:50
 */
public class _714_MaxProfit_with_fee_trade {

    public static void main(String[] args) {
        int[] prices = new int[]{1, 3, 2, 8, 4, 9};
//        int[] prices = new int[]{1,3,7,5,10,3};
        int maxProfit = new _714_MaxProfit_with_fee_trade().maxProfitDp(prices, 2);
        System.out.println(maxProfit);
        int maxProfit1 = new _714_MaxProfit_with_fee_trade().maxProfitDpImprove(prices, 2);
        System.out.println(maxProfit1);
    }

    public int maxProfitDp(int[] prices, int fee) {
        if (prices.length < 2)
            return 0;
        // 定义dp[i][0]表示第i天没有股票的最大收益，dp[i][1]表示第i天持有股票的最大收益
        int[][] dp = new int[prices.length][2];
        dp[0][0] = 0;
        dp[0][1] = 0 - prices[0];

        for (int i = 1; i < prices.length; i++) {
            dp[i][0] = Math.max(dp[i - 1][0], dp[i - 1][1] + prices[i] - fee);// 定义卖出的时候扣减手续费
            dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] - prices[i]);
        }

        return dp[prices.length - 1][0];
    }

    public int maxProfitDpImprove(int[] prices, int fee) {
        if (prices.length < 2)
            return 0;
        int maxProfitWithoutBlock = 0, maxProfitWithBlock = 0 - prices[0], tempMaxProfitWithoutStock = 0, tempMaxProfitWithStock = 0;// 定义买入的时候扣减手续费

        for (int i = 1; i < prices.length; i++) {
            tempMaxProfitWithoutStock = Math.max(maxProfitWithoutBlock, maxProfitWithBlock + prices[i] - fee);
            tempMaxProfitWithStock  = Math.max(maxProfitWithBlock, maxProfitWithoutBlock - prices[i]);

            maxProfitWithoutBlock = tempMaxProfitWithoutStock;
            maxProfitWithBlock = tempMaxProfitWithStock;
        }

        return maxProfitWithoutBlock;
    }
}
